const ws = require("nodejs-websocket");
let pub_ws = null
let sub_ws = null

ws.createServer(function (conn) {
  // 收到websocket连接
  conn.on("text", function (str) {
    console.log(str)
    if (pub_ws === conn) {
      sub_ws && sub_ws.send(str);
    } else if (sub_ws === conn) {
      pub_ws && pub_ws.send(str);
    } else {
      if (str === 'publish') {
        pub_ws = conn;
      } else {
        sub_ws = conn;
      }
    }
  })
  conn.on("error", function (event) {
    console.log('异常退出！', event)
  })
}).listen(9000)
